#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File    :   orochi_contraband.py    
@Contact :   fengfeng.qiu@amh-group.com

@Modify Time      @Author    
------------      -------    
2022/2/18 19:05   qiufengfeng      
'''

from locust import HttpUser,task,between
import json
import pandas as pd


class HelloWorldUser(HttpUser):
    wait_time = between(0.5, 1)
    host = "http://localhost:8000"

    @task
    def hello_word(self):
        import random

        data = {
            "dialogue":[
                [
                    "$$汽油$$多少钱"
                ]
            ],
            "just_predict": 1
        }
        header = {"Content-Type": "application/json"}
        import json
        self.client.post("/phantom/tf-predict",data=json.dumps(data),headers=header)






